"""
 watch -n 0.1 -x nvidia-smi   查看每0.1秒执行nvidia-smi命令的结果
 tf默认占用gpu的全部内存 因此要设置成自增长

"""
import tensorflow as tf 

tf.debugging.set_log_device_placement(True)  # 训练时显示参数在哪块gpu上

gpus = tf.config.experimental.list_physical_devices('GPU')  # 返回所有的物理gpu
for gpu in gpus:
    tf.config.experimental.set_memory_growth(True,gpu)  # 设置内存自增长：必须在程序运行前设置 
print(len(gpus))

logical_gpus = tf.config.experimental.list_logical_devices('GPU')  # 返回所有的逻辑gpu
print(len(logical_gpus))

